# -*- coding: utf-8 -*-
from datetime import datetime
from django.db import models
from main.models import Employee

grafik = {    u'Январь':u'Июнь',
              u'Февраль':u'Июль',
              u'Март':u'Октябрь',
              u'Апрель':u'Март',
              u'Май':u'Январь',
              u'Июнь':u'Февраль',
              u'Июль':u'Декабрь',
              u'Август':u'Апрель',
              u'Сентябрь':u'Ноябрь',
              u'Октябрь':u'Май',
              u'Ноябрь':u'Август',
              u'Декабрь':u'Сентябрь',
              }

class Vacation(models.Model):
    employee = models.ForeignKey(Employee, verbose_name="Сотрудник")
    a_vacation1 = models.CharField(blank=True, max_length=30, null=True, verbose_name="Рекомендация Начальника на первый отпуск")
    a_vacation2 = models.CharField(blank=True, max_length=30, null=True, verbose_name="Рекомендация Начальника на второй отпуск")
    m_vacation1 = models.CharField(blank=True, max_length=30, null=True, verbose_name="Рекомендация Мастера на первый отпуск")
    m_vacation2 = models.CharField(blank=True, max_length=30, null=True, verbose_name="Рекомендация Мастера на второй отпуск")
    year = models.CharField(blank=True, null=True, max_length=30, verbose_name="Год отпуска")
    a_vac_time = models.DateField(blank=True, null=True, verbose_name="Изменения отпуска начальником")
    m_vac_time = models.DateField(blank=True, null=True, verbose_name="Изменения отпуска мастером")

    class Meta:
        verbose_name=u'Отпуск'
        verbose_name_plural = u'Отпуска'

    def __unicode__(self):
        return u'%s' % self.employee.__unicode__()

    def get_grafik(self):
        try:
            a = Vacation.objects.get(employee=self.employee, year=datetime.now().year)
        except Vacation.DoesNotExist:
            return None
        if a.a_vacation2:
            return grafik[a.a_vacation2]
        if a.a_vacation1:
            return grafik[a.a_vacation1]
        return None
